import click
from app import db
from app.models.user import User


def register_init_commands(app):
    @app.cli.command("init-user")
    def init_user():
        """创建默认用户：用户名=周佳伟，密码=123456"""
        # 检查是否已存在该用户
        existing_user = User.query.filter_by(username="周佳伟").first()
        if existing_user:
            click.echo("用户「周佳伟」已存在，无需重复创建")
            return

        # 创建用户并设置密码
        user = User(
            username="周佳伟",
            email="zhoujiawei@example.com"  # 可自定义邮箱
        )
        user.set_password("123456")  # 密码加密存储为123456
        db.session.add(user)
        db.session.commit()
        click.echo("默认用户创建成功：用户名=周佳伟，密码=123456")

